home *** CD-ROM | disk | FTP | other *** search
/ Tech Arsenal 1 / Tech Arsenal (Arsenal Computer).ISO / tek-01 / snip0493.zip / GETSTRNG.C < prev    next >
C/C++ Source or Header  |  1993-04-05  |  2KB  |  67 lines
  1. /*
  2. ** GETSTRNG.C -- Demonstation of dynamic memory allocation to
  3. **               receive string of unknown length.
  4. **
  5. ** Ron Sires 1/31/89, released to the public domain.
  6. ** Bob Stout 2/18/93, modified to use a static buffer
  7. */
  8.  
  9. #include <stdlib.h>
  10. #include <stdio.h>
  11.  
  12. #define BLOCKSIZ    16
  13.  
  14. char *getstring(void)
  15. {
  16.       int    newchar;
  17.       size_t i;
  18.       static size_t bufsize = 0;
  19.       static char  *buffer  = NULL;
  20.  
  21.       /* Get chars from keyboard and put them in buffer.  */
  22.  
  23.       for (i = 0; ((newchar = getchar()) != EOF) && (newchar != '\n')
  24.             && (newchar != '\r'); ++i )
  25.       {
  26.             if (i + 1 > bufsize)
  27.             {
  28.                   /* If buffer is full, resize it. */
  29.  
  30.                   if (NULL == (buffer = realloc(buffer, bufsize + BLOCKSIZ)))
  31.                   {
  32.                         puts("\agetstrng() - Insufficient memory");
  33.  
  34.                         /* Add terminator to partial string */
  35.  
  36.                         if (buffer)
  37.                               buffer[i] = '\0';
  38.                         return buffer;
  39.                   }
  40.                   bufsize += BLOCKSIZ;
  41.             }
  42.             buffer[i] = (char) newchar;
  43.       }
  44.       buffer[i] = '\0';   /* Tack on a null-terminator. */
  45.       return buffer;
  46. }
  47.  
  48. #ifdef TEST
  49.  
  50. #include <string.h>
  51.  
  52. int main(void)
  53. {
  54.       char *string;
  55.  
  56.       puts("Enter strings of any length or <Enter> to quit\n");
  57.       do
  58.       {
  59.             string = getstring();
  60.             printf("You entered:\n\"%s\"\n\n", string);
  61.       } while (strlen(string));
  62.       free(string);
  63.       return EXIT_SUCCESS;
  64. }
  65.  
  66. #endif /* TEST */
  67.